numpy>=1.19.5
scikit-learn>=0.24.1
tensorflow>=2.5.0
tensorflow_datasets
ml_collections
